import moviepy.editor as mp
import time
import sys
import os

path = sys.argv[1]
fileName = sys.argv[2]
erweima = sys.argv[3]
# erweima = "qrcode.jpg"
# path = "./"
# fileName="video.mp4"
os.chdir(path)
video = mp.VideoFileClip(path+fileName)
video.save_frame("logo1.jpeg")        #保存第一帧
video.save_frame("frame1-2.png", t=2)    #保存2s时刻的那一帧
video.save_frame("frame1-4.png", t=4)    #保存2s时刻的那一帧
video.save_frame("frame1-8.png", t=8)    #保存2s时刻的那一帧
logo = (mp.ImageClip(erweima)
          .set_duration(video.duration)  # 水印持续时间
          .resize(height=150)  # 水印的高度，会等比缩放
          .margin(right=0, top=0, opacity=1)  # 水印边距和透明度
          .set_pos(("right","top")))  # 水印的位置
final = mp.CompositeVideoClip([video, logo])
final.write_videofile(path+fileName+"temp.mp4", audio=False)
time.sleep(1)
vid=mp.VideoFileClip(path+fileName+"temp.mp4")  #无声视频
audioclip =mp.AudioFileClip(path+fileName)
video = vid.set_audio(audioclip)  #音频视频合成
time.sleep(1)
vid.save_frame("logo2.jpeg")        #保存第一帧
vid.save_frame("frame2-2.png", t=2)    #保存2s时刻的那一帧
vid.save_frame("frame2-4.png", t=4)    #保存2s时刻的那一帧
vid.save_frame("frame2-8.png", t=8)    #保存2s时刻的那一帧
video.write_videofile(path+fileName+"-with-"+erweima+".mp4",audio_codec="aac")
os.system("rm "+path+fileName+"temp.mp4")
os.makedirs(fileName.split(".")[0])
os.system("mv  *.png "+"./"+fileName.split(".")[0])
os.system("mv  *.jpeg "+"./"+fileName.split(".")[0])
os.system("mv   "+path+fileName+"-with-"+erweima+".mp4 " +"./"+fileName.split(".")[0])